ValueError: logits and labels must have the same shape, received ((None, 2) vs (None, 1)).

tf.keras.layers.Dense(1, activation="sigmoid") # binary activation output

model.add(Flatten())

model.add(GlobalAveragePooling2D())